Generator function: Too many values to unpack (expected 2)